#!/bin/bash

# 定义一个函数来执行任务
run_task() {
    local aggregation=$1
    local beta=$2
    local dataset=$3
    local device=$4
    local lr=$5
    local epochs=$6
    local local_ep=$7
    local global_ep=$8

    echo "Running task: $aggregation, $beta, $dataset, $device, $lr, $epochs, $local_ep, $global_ep"
    python main.py --aggregation=$aggregation --beta=$beta --dataset=$dataset \
    --device=$device --lr=$lr --epochs=$epochs --local_ep=$local_ep --global_ep=$global_ep
}

# 执行任务列表
run_task "generator" 0.5 "mnist" "cuda:1" 0.0002 10 200 200
run_task "generator" 0.3 "mnist" "cuda:1" 0.0002 10 200 200
run_task "generator" 0.1 "mnist" "cuda:1" 0.0002 10 200 200
#
#run_task "generator" 0.5 "fmnist" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.3 "fmnist" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.1 "fmnist" "cuda:1" 0.0002 10 200 200
##
#run_task "generator" 0.5 "svhn" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.3 "svhn" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.1 "svhn" "cuda:1" 0.0002 10 200 200
##
#run_task "generator" 0.5 "cifar10" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.3 "cifar10" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.1 "cifar10" "cuda:1" 0.0002 10 200 200
##
#run_task "generator" 0.5 "cifar100" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.3 "cifar100" "cuda:1" 0.0002 10 200 200
#run_task "generator" 0.1 "cifar100" "cuda:1" 0.0002 10 200 200